package com.hsh.utils;/**
 * @author xrkhy
 * @date 2025/6/27 8:46
 * @description
 */

import java.io.InputStream;
import java.lang.reflect.Field;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * 数据库连接工具类
 */
public class MySQLConnectUtils<T> {
    private static String driver;
    private static String url;
    private static String user;
    private static String pwd;
    public static Connection conn = null;
    static ResultSet rs = null;
    static PreparedStatement ps = null;

    static {
        Properties properties = new Properties();
        ClassLoader classLoader = MySQLConnectUtils.class.getClassLoader();
        InputStream is = classLoader.getResourceAsStream("com/hsh/config/database.properties");

        try {
            properties.load(is);
            driver = properties.getProperty("driver");
            url = properties.getProperty("url");
            user = properties.getProperty("user");
            pwd = properties.getProperty("password");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取数据库连接
     */
    public static void getConnection() {
        try {
            Class.forName(driver);
            conn = DriverManager.getConnection(url, user, pwd);

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 执行查询语句 注意此方法默认返回所有结果集，且类的属性必须与结果集字段一致
     * @param sql 查询语句
     * @param params 预校验参数
     * @param clazz 封装查询结果的类
     * @return  List<T> 封装了查询结果
     * @param <T> 你要使用的数据库表的对象类型
     */
    public static <T> List<T>  executeQueryForList(String sql, Object[] params,Class<T> clazz) {
        List<T> resultList = new ArrayList<>();
        try {
            getConnection();
            ps = conn.prepareStatement(sql);
            if (params != null) {
                for (int i = 0; i < params.length; i++) {
                    ps.setObject(i + 1, params[i]);
                }
            }
            rs = ps.executeQuery();
            // 获取结果集元数据  getMetaData()存放的有 数据库表的字段名、字段类型、字段的描述信息
            ResultSetMetaData metaData = rs.getMetaData();
            // 获取数据库表 的字段数量
            int columnCount = metaData.getColumnCount();
            // 遍历结果集
            while (rs.next()) {
                // 通过反射创建对象，比如传入的是 User.class
                // 则下面这句话相当于, User user = new User();
                T obj = clazz.getDeclaredConstructor().newInstance();
                // 反射注入属性
                for (int i = 1; i <= columnCount; i++) {
                    // 获取列名 如果列有了别名 则用优先使用别名
                    String columnName = metaData.getColumnLabel(i);
                    // 获取数据库表 列的值
                    Object value = rs.getObject(i);
                    // 通过反射设置属性 将数据库表中的列的值，注入到 对象的属性中
                    Field field = clazz.getDeclaredField(columnName);
                    // 打开权限 相当于 private 属性 可以被访问
                    field.setAccessible(true);
                    // 设置 属性值 若传入的是User.class
                    // 则ob是User对象，field是User对象中的属性，value是数据库表中的列的值
                    field.set(obj, value);
                }
                resultList.add(obj);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            MySQLConnectUtils.closeAll();
        }
        if (resultList.isEmpty()){
            return null;
        }
        return resultList;
    }

    /**
     * 执行查询并返回单个结果对象，注意此方法默认返回所有结果集，且类的属性必须与结果集字段一致
     * @param sql    SQL查询语句
     * @param params SQL参数
     * @param clazz  目标对象的Class类型
     * @return 单个对象（结果为空时返回null，结果不唯一时抛异常）
     */
    public static <T> T executeQueryForObject(String sql, Object[] params, Class<T> clazz) {
        List<T> list = executeQueryForList(sql, params, clazz); // 复用现有列表查询方法
        if (list == null || list.isEmpty()) {
            return null; // 无结果时返回null
        }
        if (list.size() > 1) {
            // 结果不唯一时抛异常
            throw new RuntimeException("得到的结果集大小是：" + list.size());
        }
        return list.get(0); // 返回唯一结果
    }


    /**
     * 封装增删改方法
     *
     * @param sql    sql语句
     * @param params SQL传参数 设置？的值
     * @return 返回受影响的行数
     */
    public static int executeUpdate(String sql, Object[] params) {
        getConnection();
        int updateRows = 0;
        try {
            ps = conn.prepareStatement(sql);
            if (params != null) {
                for (int i = 0; i < params.length; i++) {
                    ps.setObject(i + 1, params[i]);
                }
                updateRows = ps.executeUpdate();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }finally {
            MySQLConnectUtils.closeAll();
        }
        return updateRows;
    }

    /**
     * 获取总记录数
     * @param sql sql语句
     * @param params 预校验参数
     * @return 总记录数
     */
    public static Integer queryTotalCount(String sql, Object[] params) {
        int totalCount = 0;
        try {
            getConnection();
            ps = conn.prepareStatement(sql);
            if (params != null) {
                for (int i = 0; i < params.length; i++) {
                    ps.setObject(i + 1, params[i]);
                }
            }
            rs = ps.executeQuery();
            if (rs.next()) {
                totalCount = rs.getInt(1);
            }
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            MySQLConnectUtils.closeAll();
        }
        return totalCount;
    }

    /**
     * 关闭连接释放资源
     */
    public static void closeAll() {
        try {
            if (conn != null) {
                conn.close();
            }
            if (ps != null) {
                ps.close();
            }
            if (rs != null) {
                rs.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}